# -*- coding: utf-8 -*-
# 用Python做运维或写自动化脚本时，难免要和文件打交道。
# 要求：输入关键字，列出指定文件夹中的所有文件名中含有此关键字的文件 / 目录，以及文件内容中含有此关键字的文件。
# 提示：建议使用os模块的listdir方法获取指定目录下的所有文件
# 主要涉及内容：文件路径、输入输出、字符串匹配

import os       # 调用os模块
def 目录():
    f = os.listdir(a)      # 获取指定目录下的所有文件
    # print(f)  # 以列表类型输出文件名称
    l = []  # 创建新列表
    for i in f:     # 遍历列表
        if ans in i:    # 查找列表中的字符有没有ans字符
            l.append(i)     # 添加到新列表
    if not l:
        print('没找到，重新输入。')
    else:
        print(l)
while True:
    a = input('输入目录\n')
    ans = input('输入关键词，回车退出。\n')
    if not ans:
        break
    目录()